import torch

x = torch.rand((3,2),dtype=torch.float32,device='cuda')
print(x.shape)
print(x.dtype)
print(x.device)